package com.neusoft.my12306.utils;

import java.security.SecureRandom;

/**
 * 验证码生成类
 * 通过随机数生成指定长度的数字验证码
 *
 * @author msprc
 */
public class GenerateRandomCodeUtil {

    /**
     * 生成默认6位长的验证码
     *
     * @return 验证码
     */
    public String createCode() {
        return createCode(6);
    }

    /**
     * 生成指定长度的随机数验证码
     *
     * @param length 需要的长度
     * @return 字符串类型的验证码
     */
    public String createCode(int length) {
        //通过循环随机数生成一个六位的验证码
        SecureRandom random = new SecureRandom();
        StringBuilder targetCode = new StringBuilder();
        for (int i = 0; i < length; i++) {
            targetCode.append(random.nextInt(9));
        }
        return targetCode.toString();
    }
}
